package com.cg.offer;

import com.cg.leetcode.TreeNode;
import org.junit.Test;

/**
 * 剑指Offer 36.二叉搜索树与双向链表
 *
 * @program: LeetCode->Offer_36
 * @description: 剑指Offer 36.二叉搜索树与双向链表
 * @author: cg
 * @create: 2022-03-23 20:00
 **/
public class Offer_36 {

    @Test
    public void test36() {
        TreeNode root = new TreeNode(4);
        root.left = new TreeNode(2);
        root.left.left = new TreeNode(1);
        root.left.right = new TreeNode(3);

        root.right = new TreeNode(5);
        System.out.println(treeToDoublyList(root));
    }

    /**
     * 输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点，只能调整树中节点指针的指向。
     * 为了让您更好地理解问题，以下面的二叉搜索树为例：
     * 我们希望将这个二叉搜索树转化为双向循环链表。链表中的每个节点都有一个前驱和后继指针。对于双向循环链表，第一个节点的前驱是最后一个节点，最后一个节点的后继是第一个节点。
     * 下图展示了上面的二叉搜索树转化成的链表。“head” 表示指向链表中有最小元素的节点。
     * 特别地，我们希望可以就地完成转换操作。当转化完成以后，树中节点的左指针需要指向前驱，树中节点的右指针需要指向后继。还需要返回链表中的第一个节点的指针。
     *
     * @param root
     * @return
     */
    public TreeNode treeToDoublyList(TreeNode root) {
        // 中序，递归解题
        // 边界值
        if (root == null) {
            return null;
        }
        dfs(root);
        // 头尾连接
        head.left = pre;
        pre.right = head;
        // 返回头节点
        return head;
    }

    TreeNode head, pre;

    public void dfs(TreeNode current) {
        // 递归结束条件
        if (current == null) {
            return;
        }
        dfs(current.left);

        // pre为空，说明是第一个结点，用head保存头结点
        if (pre == null) {
            head = current;
        } else {
            // 中间结点，pre保存的是上一个结点，那么pre的right指向当前
            pre.right = current;
        }
        // 将当前结点的left指向前结点pre
        current.left = pre;
        // 将pre后移
        pre = current;

        dfs(current.right);
    }

}
